package com.webserver.http;

import javax.swing.plaf.PanelUI;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.net.Socket;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;

/**
 * 响应对象
 * 该类的每一个实例用于表示一个HTTP的响应内容
 * 每个相应由三部分组成
 * 状态行，响应头，响应正文
 */
public class HttpResponse {
    //状态行相关信息
    private String protocol="HTTP/1.1";//协议版本
    //默认200，OK的原因是大部分响应都是正常的
    private int statusCode=200;//状态代码
    private String statusReason="OK";//状态描述

    //响应头相关信息
    //key:响应头的名字 value：响应头的值
    private Map<String,String>headers=new HashMap<>();

    // 响应正文相关信息
    private File entity;//正文对应的一个实际文件

    private Socket socket;
    public HttpResponse(Socket socket) {
        this.socket=socket;
    }
    /**
     * 将当前响应对象内容按照标准的HTTP响应格式发送给客户端
     */
    public void flush() throws IOException {
        //3.1发送状态行
        sendStatusLine();
        //3.2:发送响应头
        sendHeaders();
        //3.3：发送正文（index页面内容）
        sendContent();
    }

    private void sendStatusLine() throws IOException {
        //UnsupportedEncodingException 原因：字符集名字拼写错误！
        String line=protocol+" "+statusCode +" "+statusReason;
       println(line);
    }
    //响应头
    private void sendHeaders() throws IOException {
//        String line = "Content-Type: text/html";
//        println(line);
//
//        line = "Content-Length: "+entity.length();
//        println(line);
        /*
            headers:
                key             value
            Content-Type        text/html
            Content-Length      12314
            Server              WebServer

         */
        Set<Map.Entry<String,String>> entrySet=headers.entrySet();
        for(Map.Entry<String,String>e:entrySet){
            String name=e.getKey();//key就是响应头的名字
            String value=e.getValue();//value就是响应头对应的值
            String line=name+": "+value;
            println(line);
        }

        //单独发送CRLF表示响应头发送完毕
      println("");
    }
    //响应正文
    private void sendContent() throws IOException {
        OutputStream out =socket.getOutputStream();

        try(FileInputStream fis=new FileInputStream(entity);
        ) {
            byte[]buf=new byte[1024*8];
            int len;
            while ((len=fis.read(buf))!=-1){
                out.write(buf,0,len);
            }
        }
    }

    private  void println(String line) throws IOException {
        OutputStream out =socket.getOutputStream();
        byte[]data=line.getBytes("ISO8859-1");
        out.write(data);
        out.write(13);//written CR
        out.write(10);//written LF
    }

    public String getProtocol() {
        return protocol;
    }

    public void setProtocol(String protocol) {
        this.protocol = protocol;
    }

    public int getStatusCode() {
        return statusCode;
    }

    public void setStatusCode(int statusCode) {
        this.statusCode = statusCode;
    }

    public String getStatusReason() {
        return statusReason;
    }

    public void setStatusReason(String statusReason) {
        this.statusReason = statusReason;
    }

    public File getEntity() {
        return entity;
    }

    public void setEntity(File entity) {
        this.entity = entity;

    }

    /**
     * 添加一个要发送的响应头
     * @param name
     * @param value
     */
    public void addHeader(String name ,String value){
        this.headers.put(name,value);
    }
}
